"""
Copyright (c) 2024 [XKW.Beijing]
All rights reserved.

Author: [tangxiaojun]
Email: [417281862@qq.com]
"""

import fasttext

# 使用FastText训练的词向量模型 cc.zh.300.bin.gz 进行测试，7G
class CczhResource:

    def load_model(self):
        self.model = fasttext.load_model('/Users/tangxiaojun/Downloads/cc.zh.300.bin')

    def get_nearest_neighbors(self, predictions_text):
        """
        获取预测文本的最近邻邻居。距离相近的词

        参数:
        - predictions_text: 预测文本，用于模型查询。

        返回值:
        - result: 查询结果，返回模型认为与预测文本最接近的邻居们。
        """
        result = self.model.get_nearest_neighbors(predictions_text)  # 查询模型以获取预测文本的最近邻邻居
        return result
